import pandas as pd

from 单用户中断时长工作.Input import get_company


# 获取到指定日期平均fttr用户
def get_fttr_user(d=100):
    df = pd.read_excel("D:\\中断输入\\活跃用户数新.xlsx", sheet_name='fttr用户')
    n = df.iloc[:, 1:d].mean(axis=1)
    if (d > len(df.columns)):
        print("活跃用户数未更新")
    df['fttr用户数'] = n
    return df[['fttr用户数', '分公司']]


def f1():
    data = {
        '分公司': ['平谷', '中区', '朝阳', '房山', '昌平', '大兴', '通州', '经开', '顺义', '丰台', '海淀', '延庆',
                   '石门', '密云', '怀柔'],
        '软探针活跃': [65554, 222348, 479501, 190403, 217060, 183438, 228129, 116885, 208073, 281090, 285173, 45553,
                       134656, 75736, 51483],
        '软探针FTTR活跃': [4058, 13715, 30252, 10191, 11785, 11030, 11350, 7575, 11206, 16784, 19229, 3078, 7394, 4345,
                           4121]
    }

    df = pd.DataFrame(data)
    company = get_company()
    df = company.merge(df, on="分公司", how='left')
    df.to_csv("/temp/t.csv")


def f2():
    # 首先将数据读入 DataFrame
    data = """房山分公司,197954
顺义分公司,214854
经开分公司,120407
大兴分公司,188738
丰台分公司,290672
通州分公司,234425
海淀分公司,293894
密云分公司,77605
延庆分公司,46831
朝阳分公司,496055
中区分公司,230524
石门分公司,138217
怀柔分公司,52998
平谷分公司,67950
昌平分公司,224481"""
    str = """房山分公司,11225
顺义分公司,11808
经开分公司,6756
大兴分公司,10577
丰台分公司,17627
通州分公司,11683
海淀分公司,20323
延庆分公司,3752
密云分公司,4492
朝阳分公司,30177
中区分公司,13713
石门分公司,7204
怀柔分公司,4905
平谷分公司,4165
昌平分公司,13089"""
    # 将数据转换为 DataFrame
    df = pd.DataFrame([line.split(',') for line in data.split('\n')], columns=['分公司', '用户数'])
    df2 = pd.DataFrame([line.split(',') for line in str.split('\n')], columns=['分公司', 'fttr用户数'])
    df['分公司'] = df['分公司'].str.lstrip()
    df2['分公司'] = df2['分公司'].str.lstrip()
    df['分公司'] = df['分公司'].str[:2]
    df2['分公司'] = df2['分公司'].str[:2]
    df['用户数'] = df['用户数'].astype(int)
    df2['fttr用户数'] = df2['fttr用户数'].astype(int)
    get_company().merge(df.merge(df2, on='分公司'), on='分公司', how='left').to_excel(
        "D:\\中断输入\\活跃用户数\\3.9活跃用户数.xlsx")


if __name__ == '__main__':
    f2()
